import os

def count_images():
    # 输出目录
    output_dir = "output_images"
    
    # 确保目录存在
    if not os.path.exists(output_dir):
        print(f"目录 {output_dir} 不存在")
        return
    
    # 支持的图片格式
    image_extensions = ('.png', '.jpg', '.jpeg', '.bmp', '.gif')
    
    # 统计总数
    total_count = 0
    # 统计每个子目录的数量
    dir_counts = {}
    
    # 遍历所有子目录
    for root, dirs, files in os.walk(output_dir):
        # 获取当前目录名（相对于output_dir）
        rel_path = os.path.relpath(root, output_dir)
        if rel_path == '.':  # 如果是根目录
            dir_name = "根目录"
        else:
            dir_name = rel_path
            
        # 统计当前目录下的图片数量
        count = sum(1 for f in files if f.lower().endswith(image_extensions))
        if count > 0:  # 只记录有图片的目录
            dir_counts[dir_name] = count
            total_count += count
    
    # 打印统计结果
    print("\n图片统计结果：")
    print("-" * 30)
    for dir_name, count in dir_counts.items():
        print(f"{dir_name}: {count} 张图片")
    print("-" * 30)
    print(f"总计: {total_count} 张图片")

if __name__ == "__main__":
    count_images() 